--------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------
--------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------
vendor/lxmfy/tests/test_external_cogs.py fee9e3356259c967b89373cf94deec01e6a5e6b3 (fee9e335) Text, 6.18 KB
Tff7b72import T7ee787os
Tff7b72import T7ee787stat
Tff7b72import T7ee787tempfile
Tff7b72from T7ee787pathlib Tff7b72import Te6edf3Path
Tff7b72from T7ee787unittestT7ee787.T7ee787mock Tff7b72import Te6edf3MagicMockTb4b4b4, Te6edf3patch
Tff7b72import T7ee787pytest
Tff7b72from T7ee787lxmfy Tff7b72import Te6edf3BotConfigTb4b4b4, Te6edf3LXMFBot
Tff7b72from T7ee787lxmfyT7ee787.T7ee787cogs_core Tff7b72import Te6edf3_get_sandbox_commandTb4b4b4, Te6edf3load_cogs_from_directory
Tf0883e@pytestTff7b72.Td2a8fffixture
Tff7b72def Td2a8ffexternal_cog_setupTb4b4b4(Tb4b4b4)Tb4b4b4:
T8b949e"""Set up a temporary directory with an external script cog."""
Tff7b72with Te6edf3tempfileTff7b72.Td2a8ffTemporaryDirectoryTb4b4b4(Tb4b4b4) Tff7b72as Te6edf3temp_dirTb4b4b4:
Te6edf3temp_path Tff7b72= Te6edf3PathTb4b4b4(Te6edf3temp_dirTb4b4b4)
Te6edf3cogs_dir Tff7b72= Te6edf3temp_path Tff7b72/ Ta5d6ff"Ta5d6ffcogsTa5d6ff"
Te6edf3cogs_dirTff7b72.Td2a8ffmkdirTb4b4b4(Tb4b4b4)
T8b949e# Create a simple bash script
Te6edf3script_path Tff7b72= Te6edf3cogs_dir Tff7b72/ Ta5d6ff"Ta5d6fftest_cmd.shTa5d6ff"
Te6edf3script_content Tff7b72= Ta5d6ff"""Ta5d6ff#!/bin/bash
Ta5d6ffecho Ta5d6ff"Ta5d6ffHello from external script! Sender: $1 Content: $2Ta5d6ff"
Ta5d6ff"""
Te6edf3script_pathTff7b72.Td2a8ffwrite_textTb4b4b4(Te6edf3script_contentTb4b4b4)
T8b949e# Make it executable
Te6edf3st Tff7b72= Te6edf3osTff7b72.Td2a8ffstatTb4b4b4(Te6edf3script_pathTb4b4b4)
Te6edf3osTff7b72.Td2a8ffchmodTb4b4b4(Te6edf3script_pathTb4b4b4, Te6edf3stTff7b72.Td2a8ffst_mode Tff7b72| Te6edf3statTff7b72.Td2a8ffS_IEXECTb4b4b4)
Tff7b72yield Te6edf3temp_path
Tff7b72def Td2a8fftest_cog_external_loadingTb4b4b4(Te6edf3external_cog_setupTb4b4b4)Tb4b4b4:
T8b949e"""Test that external script cogs are loaded correctly."""
Te6edf3config Tff7b72= Te6edf3BotConfigTb4b4b4(
Te6edf3nameTff7b72=Ta5d6ff"Ta5d6ffTestBotTa5d6ff"Tb4b4b4,
Te6edf3test_modeTff7b72=Tff7b72TrueTb4b4b4,
Te6edf3external_cogs_enabledTff7b72=Tff7b72TrueTb4b4b4,
Te6edf3config_pathTff7b72=Tffa657strTb4b4b4(Te6edf3external_cog_setupTb4b4b4)Tb4b4b4,
Te6edf3storage_pathTff7b72=Tffa657strTb4b4b4(Te6edf3external_cog_setup Tff7b72/ Ta5d6ff"Ta5d6ffstorageTa5d6ff"Tb4b4b4)Tb4b4b4,
Tb4b4b4)
Te6edf3bot Tff7b72= Te6edf3LXMFBotTb4b4b4(Tff7b72*Tff7b72*Te6edf3configTff7b72.Td2a8ff__dict__Tb4b4b4)
Te6edf3botTff7b72.Td2a8ffconfig_path Tff7b72= Tffa657strTb4b4b4(Te6edf3external_cog_setupTb4b4b4)
T8b949e# Load cogs
Te6edf3load_cogs_from_directoryTb4b4b4(Te6edf3botTb4b4b4)
Tff7b72assert Ta5d6ff"Ta5d6fftest_cmdTa5d6ff" Tff7b72in Te6edf3botTff7b72.Td2a8ffcommands
Te6edf3cmd Tff7b72= Te6edf3botTff7b72.Td2a8ffcommandsTb4b4b4[Ta5d6ff"Ta5d6fftest_cmdTa5d6ff"Tb4b4b4]
Tff7b72assert Te6edf3cmdTff7b72.Td2a8ffname Tff7b72== Ta5d6ff"Ta5d6fftest_cmdTa5d6ff"
Tff7b72assert Te6edf3cmdTff7b72.Td2a8ffthreaded Tff7b72is Tff7b72True T8b949e# Should always be threaded
Tff7b72assert Ta5d6ff"Ta5d6ffExternal script commandTa5d6ff" Tff7b72in Te6edf3cmdTff7b72.Td2a8ffdescription
Tff7b72def Td2a8fftest_cog_external_executionTb4b4b4(Te6edf3external_cog_setupTb4b4b4)Tb4b4b4:
T8b949e"""Test that external script cogs execute and return output."""
Te6edf3config Tff7b72= Te6edf3BotConfigTb4b4b4(
Te6edf3nameTff7b72=Ta5d6ff"Ta5d6ffTestBotTa5d6ff"Tb4b4b4,
Te6edf3test_modeTff7b72=Tff7b72TrueTb4b4b4,
Te6edf3external_cogs_enabledTff7b72=Tff7b72TrueTb4b4b4,
Te6edf3external_cogs_sandbox_enabledTff7b72=Tff7b72FalseTb4b4b4,
Te6edf3config_pathTff7b72=Tffa657strTb4b4b4(Te6edf3external_cog_setupTb4b4b4)Tb4b4b4,
Te6edf3storage_pathTff7b72=Tffa657strTb4b4b4(Te6edf3external_cog_setup Tff7b72/ Ta5d6ff"Ta5d6ffstorageTa5d6ff"Tb4b4b4)Tb4b4b4,
Tb4b4b4)
Te6edf3bot Tff7b72= Te6edf3LXMFBotTb4b4b4(Tff7b72*Tff7b72*Te6edf3configTff7b72.Td2a8ff__dict__Tb4b4b4)
Te6edf3botTff7b72.Td2a8ffconfig_path Tff7b72= Tffa657strTb4b4b4(Te6edf3external_cog_setupTb4b4b4)
T8b949e# Load cogs
Te6edf3load_cogs_from_directoryTb4b4b4(Te6edf3botTb4b4b4)
Te6edf3cmd Tff7b72= Te6edf3botTff7b72.Td2a8ffcommandsTb4b4b4[Ta5d6ff"Ta5d6fftest_cmdTa5d6ff"Tb4b4b4]
T8b949e# Mock message
Te6edf3msg Tff7b72= Te6edf3MagicMockTb4b4b4(Tb4b4b4)
Te6edf3msgTff7b72.Td2a8ffsender Tff7b72= Ta5d6ff"Ta5d6fftest_senderTa5d6ff"
Te6edf3msgTff7b72.Td2a8ffcontent Tff7b72= Ta5d6ff"Ta5d6ff/test_cmd hello worldTa5d6ff"
Te6edf3msgTff7b72.Td2a8ffargs Tff7b72= Tb4b4b4[Ta5d6ff"Ta5d6ffhelloTa5d6ff"Tb4b4b4, Ta5d6ff"Ta5d6ffworldTa5d6ff"Tb4b4b4]
T8b949e# Execute callback
Te6edf3cmdTff7b72.Td2a8ffcallbackTb4b4b4(Te6edf3msgTb4b4b4)
T8b949e# Check if msg.reply was called with expected output
Te6edf3msgTff7b72.Td2a8ffreplyTff7b72.Td2a8ffassert_called_onceTb4b4b4(Tb4b4b4)
Te6edf3argsTb4b4b4, Te6edf3_ Tff7b72= Te6edf3msgTff7b72.Td2a8ffreplyTff7b72.Td2a8ffcall_args
Te6edf3output Tff7b72= Te6edf3argsTb4b4b4[T79c0ff0Tb4b4b4]
Tff7b72assert Ta5d6ff"Ta5d6ffHello from external script!Ta5d6ff" Tff7b72in Te6edf3output
Tff7b72assert Ta5d6ff"Ta5d6ffSender: test_senderTa5d6ff" Tff7b72in Te6edf3output
Tff7b72assert Ta5d6ff"Ta5d6ffContent: /test_cmd hello worldTa5d6ff" Tff7b72in Te6edf3output
Tff7b72def Td2a8fftest_cog_external_disabledTb4b4b4(Te6edf3external_cog_setupTb4b4b4)Tb4b4b4:
T8b949e"""Test that external script cogs are NOT loaded when disabled."""
Te6edf3config Tff7b72= Te6edf3BotConfigTb4b4b4(
Te6edf3nameTff7b72=Ta5d6ff"Ta5d6ffTestBotTa5d6ff"Tb4b4b4,
Te6edf3test_modeTff7b72=Tff7b72TrueTb4b4b4,
Te6edf3external_cogs_enabledTff7b72=Tff7b72FalseTb4b4b4,
Te6edf3config_pathTff7b72=Tffa657strTb4b4b4(Te6edf3external_cog_setupTb4b4b4)Tb4b4b4,
Te6edf3storage_pathTff7b72=Tffa657strTb4b4b4(Te6edf3external_cog_setup Tff7b72/ Ta5d6ff"Ta5d6ffstorageTa5d6ff"Tb4b4b4)Tb4b4b4,
Tb4b4b4)
Te6edf3bot Tff7b72= Te6edf3LXMFBotTb4b4b4(Tff7b72*Tff7b72*Te6edf3configTff7b72.Td2a8ff__dict__Tb4b4b4)
Te6edf3botTff7b72.Td2a8ffconfig_path Tff7b72= Tffa657strTb4b4b4(Te6edf3external_cog_setupTb4b4b4)
T8b949e# Load cogs
Te6edf3load_cogs_from_directoryTb4b4b4(Te6edf3botTb4b4b4)
Tff7b72assert Ta5d6ff"Ta5d6fftest_cmdTa5d6ff" Tff7b72not Tff7b72in Te6edf3botTff7b72.Td2a8ffcommands
Tff7b72def Td2a8fftest_sandbox_detection_bwrapTb4b4b4(Tb4b4b4)Tb4b4b4:
T8b949e"""Test bubblewrap sandbox detection."""
Te6edf3bot Tff7b72= Te6edf3MagicMockTb4b4b4(Tb4b4b4)
Te6edf3botTff7b72.Td2a8ffconfigTff7b72.Td2a8ffexternal_cogs_sandbox_enabled Tff7b72= Tff7b72True
Te6edf3botTff7b72.Td2a8ffconfigTff7b72.Td2a8ffexternal_cogs_sandbox_type Tff7b72= Ta5d6ff"Ta5d6ffautoTa5d6ff"
Tff7b72with Tb4b4b4(
Te6edf3patchTb4b4b4(Ta5d6ff"Ta5d6ffshutil.whichTa5d6ff"Tb4b4b4) Tff7b72as Te6edf3mock_whichTb4b4b4,
Te6edf3patchTb4b4b4(Ta5d6ff"Ta5d6ffsys.platformTa5d6ff"Tb4b4b4, Ta5d6ff"Ta5d6fflinuxTa5d6ff"Tb4b4b4)Tb4b4b4,
Te6edf3patchTb4b4b4(Ta5d6ff"Ta5d6ffos.path.existsTa5d6ff"Tb4b4b4, Te6edf3return_valueTff7b72=Tff7b72TrueTb4b4b4)Tb4b4b4,
Tb4b4b4)Tb4b4b4:
Te6edf3mock_whichTff7b72.Td2a8ffside_effect Tff7b72= Tff7b72lambda Te6edf3xTb4b4b4: Ta5d6fffTa5d6ff"Ta5d6ff/usr/bin/Tffd700{Te6edf3xTffd700}Ta5d6ff" Tff7b72if Te6edf3x Tff7b72== Ta5d6ff"Ta5d6ffbwrapTa5d6ff" Tff7b72else Tff7b72None
Te6edf3cmd Tff7b72= Te6edf3_get_sandbox_commandTb4b4b4(Te6edf3botTb4b4b4, Ta5d6ff"Ta5d6ff/path/to/scriptTa5d6ff"Tb4b4b4)
Tff7b72assert Te6edf3cmd Tff7b72is Tff7b72not Tff7b72None
Tff7b72assert Ta5d6ff"Ta5d6ff/usr/bin/bwrapTa5d6ff" Tff7b72in Te6edf3cmd
Tff7b72def Td2a8fftest_sandbox_detection_firejailTb4b4b4(Tb4b4b4)Tb4b4b4:
T8b949e"""Test firejail sandbox detection."""
Te6edf3bot Tff7b72= Te6edf3MagicMockTb4b4b4(Tb4b4b4)
Te6edf3botTff7b72.Td2a8ffconfigTff7b72.Td2a8ffexternal_cogs_sandbox_enabled Tff7b72= Tff7b72True
Te6edf3botTff7b72.Td2a8ffconfigTff7b72.Td2a8ffexternal_cogs_sandbox_type Tff7b72= Ta5d6ff"Ta5d6ffautoTa5d6ff"
Tff7b72with Te6edf3patchTb4b4b4(Ta5d6ff"Ta5d6ffshutil.whichTa5d6ff"Tb4b4b4) Tff7b72as Te6edf3mock_whichTb4b4b4, Te6edf3patchTb4b4b4(Ta5d6ff"Ta5d6ffsys.platformTa5d6ff"Tb4b4b4, Ta5d6ff"Ta5d6fflinuxTa5d6ff"Tb4b4b4)Tb4b4b4:
Te6edf3mock_whichTff7b72.Td2a8ffside_effect Tff7b72= Tff7b72lambda Te6edf3xTb4b4b4: Ta5d6fffTa5d6ff"Ta5d6ff/usr/bin/Tffd700{Te6edf3xTffd700}Ta5d6ff" Tff7b72if Te6edf3x Tff7b72== Ta5d6ff"Ta5d6fffirejailTa5d6ff" Tff7b72else Tff7b72None
Te6edf3cmd Tff7b72= Te6edf3_get_sandbox_commandTb4b4b4(Te6edf3botTb4b4b4, Ta5d6ff"Ta5d6ff/path/to/scriptTa5d6ff"Tb4b4b4)
Tff7b72assert Te6edf3cmd Tff7b72is Tff7b72not Tff7b72None
Tff7b72assert Ta5d6ff"Ta5d6ff/usr/bin/firejailTa5d6ff" Tff7b72in Te6edf3cmd
Tff7b72def Td2a8fftest_cog_external_timeout_enforcementTb4b4b4(Te6edf3external_cog_setupTb4b4b4)Tb4b4b4:
T8b949e"""Test that external script cogs time out correctly."""
T8b949e# Create a script that sleeps forever
Te6edf3cogs_dir Tff7b72= Te6edf3external_cog_setup Tff7b72/ Ta5d6ff"Ta5d6ffcogsTa5d6ff"
Te6edf3timeout_script Tff7b72= Te6edf3cogs_dir Tff7b72/ Ta5d6ff"Ta5d6fftimeout.shTa5d6ff"
Te6edf3timeout_scriptTff7b72.Td2a8ffwrite_textTb4b4b4(Ta5d6ff"Ta5d6ff#!/bin/bashTffea00\nTa5d6ffsleep 10Ta5d6ff"Tb4b4b4)
Te6edf3osTff7b72.Td2a8ffchmodTb4b4b4(Te6edf3timeout_scriptTb4b4b4, Te6edf3osTff7b72.Td2a8ffstatTb4b4b4(Te6edf3timeout_scriptTb4b4b4)Tff7b72.Td2a8ffst_mode Tff7b72| Te6edf3statTff7b72.Td2a8ffS_IEXECTb4b4b4)
Te6edf3config Tff7b72= Te6edf3BotConfigTb4b4b4(
Te6edf3nameTff7b72=Ta5d6ff"Ta5d6ffTestBotTa5d6ff"Tb4b4b4,
Te6edf3test_modeTff7b72=Tff7b72TrueTb4b4b4,
Te6edf3external_cogs_enabledTff7b72=Tff7b72TrueTb4b4b4,
Te6edf3external_cogs_sandbox_enabledTff7b72=Tff7b72FalseTb4b4b4,
Te6edf3external_cogs_timeoutTff7b72=T79c0ff1Tb4b4b4, T8b949e# 1 second timeout
Te6edf3config_pathTff7b72=Tffa657strTb4b4b4(Te6edf3external_cog_setupTb4b4b4)Tb4b4b4,
Te6edf3storage_pathTff7b72=Tffa657strTb4b4b4(Te6edf3external_cog_setup Tff7b72/ Ta5d6ff"Ta5d6ffstorageTa5d6ff"Tb4b4b4)Tb4b4b4,
Tb4b4b4)
Te6edf3bot Tff7b72= Te6edf3LXMFBotTb4b4b4(Tff7b72*Tff7b72*Te6edf3configTff7b72.Td2a8ff__dict__Tb4b4b4)
Te6edf3botTff7b72.Td2a8ffconfig_path Tff7b72= Tffa657strTb4b4b4(Te6edf3external_cog_setupTb4b4b4)
Te6edf3load_cogs_from_directoryTb4b4b4(Te6edf3botTb4b4b4)
Te6edf3cmd Tff7b72= Te6edf3botTff7b72.Td2a8ffcommandsTb4b4b4[Ta5d6ff"Ta5d6fftimeoutTa5d6ff"Tb4b4b4]
Te6edf3msg Tff7b72= Te6edf3MagicMockTb4b4b4(Tb4b4b4)
Te6edf3cmdTff7b72.Td2a8ffcallbackTb4b4b4(Te6edf3msgTb4b4b4)
Te6edf3msgTff7b72.Td2a8ffreplyTff7b72.Td2a8ffassert_called_withTb4b4b4(Ta5d6ff"Ta5d6ffError: Command timeout.sh timed out.Ta5d6ff"Tb4b4b4)
Tff7b72def Td2a8fftest_cog_external_timeout_disabledTb4b4b4(Te6edf3external_cog_setupTb4b4b4)Tb4b4b4:
T8b949e"""Test that external script cogs don't time out when timeout is 0."""
T8b949e# Create a script that sleeps for 2 seconds
Te6edf3cogs_dir Tff7b72= Te6edf3external_cog_setup Tff7b72/ Ta5d6ff"Ta5d6ffcogsTa5d6ff"
Te6edf3sleep_script Tff7b72= Te6edf3cogs_dir Tff7b72/ Ta5d6ff"Ta5d6ffsleep.shTa5d6ff"
Te6edf3sleep_scriptTff7b72.Td2a8ffwrite_textTb4b4b4(Ta5d6ff"Ta5d6ff#!/bin/bashTffea00\nTa5d6ffsleep 1.5Tffea00\nTa5d6ffecho Ta5d6ff'Ta5d6ffDone sleepingTa5d6ff'Ta5d6ff"Tb4b4b4)
Te6edf3osTff7b72.Td2a8ffchmodTb4b4b4(Te6edf3sleep_scriptTb4b4b4, Te6edf3osTff7b72.Td2a8ffstatTb4b4b4(Te6edf3sleep_scriptTb4b4b4)Tff7b72.Td2a8ffst_mode Tff7b72| Te6edf3statTff7b72.Td2a8ffS_IEXECTb4b4b4)
Te6edf3config Tff7b72= Te6edf3BotConfigTb4b4b4(
Te6edf3nameTff7b72=Ta5d6ff"Ta5d6ffTestBotTa5d6ff"Tb4b4b4,
Te6edf3test_modeTff7b72=Tff7b72TrueTb4b4b4,
Te6edf3external_cogs_enabledTff7b72=Tff7b72TrueTb4b4b4,
Te6edf3external_cogs_sandbox_enabledTff7b72=Tff7b72FalseTb4b4b4,
Te6edf3external_cogs_timeoutTff7b72=T79c0ff0Tb4b4b4, T8b949e# Infinite timeout
Te6edf3config_pathTff7b72=Tffa657strTb4b4b4(Te6edf3external_cog_setupTb4b4b4)Tb4b4b4,
Te6edf3storage_pathTff7b72=Tffa657strTb4b4b4(Te6edf3external_cog_setup Tff7b72/ Ta5d6ff"Ta5d6ffstorageTa5d6ff"Tb4b4b4)Tb4b4b4,
Tb4b4b4)
Te6edf3bot Tff7b72= Te6edf3LXMFBotTb4b4b4(Tff7b72*Tff7b72*Te6edf3configTff7b72.Td2a8ff__dict__Tb4b4b4)
Te6edf3botTff7b72.Td2a8ffconfig_path Tff7b72= Tffa657strTb4b4b4(Te6edf3external_cog_setupTb4b4b4)
Te6edf3load_cogs_from_directoryTb4b4b4(Te6edf3botTb4b4b4)
Te6edf3cmd Tff7b72= Te6edf3botTff7b72.Td2a8ffcommandsTb4b4b4[Ta5d6ff"Ta5d6ffsleepTa5d6ff"Tb4b4b4]
Te6edf3msg Tff7b72= Te6edf3MagicMockTb4b4b4(Tb4b4b4)
Te6edf3cmdTff7b72.Td2a8ffcallbackTb4b4b4(Te6edf3msgTb4b4b4)
Te6edf3msgTff7b72.Td2a8ffreplyTff7b72.Td2a8ffassert_called_withTb4b4b4(Ta5d6ff"Ta5d6ffDone sleepingTa5d6ff"Tb4b4b4)
──────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────